package com.lu.manage.modular.system.websocket;

import com.alibaba.fastjson.JSON;
import com.lu.manage.modular.system.model.server.Server;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Component;

import javax.websocket.*;
import javax.websocket.server.PathParam;
import javax.websocket.server.ServerEndpoint;
import java.io.IOException;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

/**
 * @Auther: ZhangLu
 * @Date: 2019/2/27 09:41
 * @DescrdeptIdtion:
 * 虽然@Component默认是单例模式的
 * ，但springboot还是会为每个websocket连接初始化一个bean。
 * ServerEndpoint它的功能主要是将目前的类定义成
 * 一个websocket服务器端。注解的值将被用于监听用户连接的终端访问URL地址。
 */
@Component
@ServerEndpoint("/system/ws/server/{deptId}/{ip}")
public class ServerMonitort {

    private static final Logger LOGGER = LoggerFactory.getLogger(ServerMonitort.class);

    /**
     * @auther: ZhangLu
     * @description 存活的session集合
     */
    private static Map<String, ServerMonitort> sessionMap = new ConcurrentHashMap<>();

    /**
     * @auther: ZhangLu
     * @description 存活的session集合
     */
    private static Map<String, ServerMonitort> ipSessionMap = new ConcurrentHashMap<>();

//    private static Object o = SpringContextHolder.getBean("IArticleService");

    // 某个客户端的连接会话，需要通过它来给客户端发送信息
    private Session session;

    // 客户端的deptId地址
    private String deptId;

    /**
     * @auther: ZhangLu
     * @param session 与客户端的websocket连接会话
     * @param deptId websocket支持路径参数
     * @date: 2019/2/27 9:50
     * @descrdeptIdtion 建立连接的回调方法，与前台onOpen相对应
     * @return: void
     */
    @OnOpen
    public void onOpen(Session session, @PathParam("deptId") String deptId,@PathParam("ip") String ip){
        LOGGER.info("》》》》》》》》》》》》》》》》》》》》》》》\r\n");
        this.session = session;
        this.deptId = deptId;
        if(!sessionMap.containsKey(deptId)){
            sessionMap.put(deptId,this);
        }
        if(!ipSessionMap.containsKey(ip)){
            ipSessionMap.put(ip,this);
        }
    }

    /**
     *
     * @param message 客户端传来的信息
     * @param session 对应的session
     * @param deptId
     * @date: 2019/2/27 9:58
     *
     * @return: void
     */
    /**
     * @auther: ZhangLu
     * @descrdeptIdtion 收到客户信息的回调方法
     * @param message  Map<String,Object> k1:title 标题 k2:message 消息
     * @param session
     * @param deptId
     */
    @OnMessage
    public void onMessage(String message,Session session,@PathParam("deptId") String deptId){
        LOGGER.info("deptId - {}  接收到信息 {}",this.deptId,message);

        // todo 客户端传来什么信息暂时不知，先放着
        Server server = Server.cpuMemJvm();
        sessionMap.forEach((k,v) -> {
            try {
                v.getSession().getBasicRemote().sendText(JSON.toJSONString(server));
            } catch (IOException e) {
                e.printStackTrace();
            }
        });

    }

    /**
     * @auther: ZhangLu
     * @param session
     * @param error 错误
     * @date: 2019/2/27 10:01
     * @descrdeptIdtion
     * @return: void
     */
    @OnError
    public void onError(Session session,Throwable error,@PathParam("deptId") String deptId){
        LOGGER.info("deptId - {} webSocket调用出错",deptId);
        // todo 出错后如何处理
        error.printStackTrace();
    }

    /**
     * @auther: ZhangLu
     * @param session 1
     * @param deptId 2
     * @date: 2019/2/27 10:04
     * @descrdeptIdtion 关闭连接,与前台的onClose相对应
     * @return: void
     */
    @OnClose
    public void onClose(Session session,@PathParam("deptId") String deptId){
            sessionMap.remove(deptId);
        // todo 退出需要去补充什么
        LOGGER.info("deptId - {}  关闭session",deptId);
        LOGGER.info("《《《《《《《《《《《《《《《《《《《《《《《\r\n");
    }

    public static Map<String, ServerMonitort> getSessionMap() {
        return sessionMap;
    }

    public static Map<String, ServerMonitort> getIpSessionMap() {
        return ipSessionMap;
    }

    public Session getSession() {
        return session;
    }

    public String getdeptId() {
        return deptId;
    }

}